package com.taldh.algorithm.listnode;

import com.taldh.algorithm.base.ListNode;
import org.junit.Assert;
import org.junit.Test;

/**
 * @author donghaoyuan
 */
public class AddTwoNumbersSolution {

    /**
     * 给出两个 非空 的链表用来表示两个非负的整数。其中，它们各自的位数是按照 逆序 的方式存储的，并且它们的每个节点只能存储 一位 数字。
     *
     * 如果，我们将这两个数相加起来，则会返回一个新的链表来表示它们的和。
     *
     * 您可以假设除了数字 0 之外，这两个数都不会以 0 开头。
     *
     * 示例：
     *
     * 输入：(2 -> 4 -> 3) + (5 -> 6 -> 4)
     * 输出：7 -> 0 -> 8
     * 原因：342 + 465 = 807
     *
     * 来源：力扣（LeetCode）
     * 链接：https://leetcode-cn.com/problems/add-two-numbers
     * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
     */
    public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
        // 1. 顺序遍历，栈抛出,
        // 时间复杂度：O(\max(m, n))O(max(m,n))，假设 mm 和 nn 分别表示 l1l1 和 l2l2 的长度，上面的算法最多重复 \max(m, n)max(m,n) 次。
        //
        //空间复杂度：O(\max(m, n))O(max(m,n))， 新列表的长度最多为 \max(m,n) + 1max(m,n)+1。
//        Stack<Integer> stack = new Stack<>();
//        ListNode result = null;
//        int base = 0;
//        while (l1 != null || l2 != null || base > 0) {
//            int bitSum = -1;
//            if (l1 != null && l2 != null) {
//                bitSum = l1.val + l2.val + base;
//                base =  bitSum / 10;
//                bitSum = bitSum % 10;
//                l1 = l1.next;
//                l2 = l2.next;
//            } else if (l1 != null) {
//                bitSum = l1.val + base;
//                base =  bitSum / 10;
//                bitSum = bitSum % 10;
//                l1 = l1.next;
//            } else if (l2 != null){
//                bitSum = l2.val + base;
//                base =  bitSum / 10;
//                bitSum = bitSum % 10;
//                l2 = l2.next;
//            } else {
//                bitSum = base;
//                base = 0;
//            }
//
//            stack.push(bitSum);
//        }
//
//        while (!stack.isEmpty()) {
//            Integer integer = stack.pop();
//            if (result == null) {
//                result = new ListNode(integer);
//            } else {
//                ListNode temp = result;
//                result = new ListNode(integer);
//                result.next = temp;
//            }
//        }
//
//        return result;
        // 优化：一个指针指向指针头
//        ListNode header = null;
//        ListNode current = null;
//        int base = 0;
//        while (l1 != null || l2 != null || base > 0) {
//            int bitSum = -1;
//            if (l1 != null && l2 != null) {
//                bitSum = l1.val + l2.val + base;
//                base =  bitSum / 10;
//                bitSum = bitSum % 10;
//                l1 = l1.next;
//                l2 = l2.next;
//            } else if (l1 != null) {
//                bitSum = l1.val + base;
//                base =  bitSum / 10;
//                bitSum = bitSum % 10;
//                l1 = l1.next;
//            } else if (l2 != null){
//                bitSum = l2.val + base;
//                base =  bitSum / 10;
//                bitSum = bitSum % 10;
//                l2 = l2.next;
//            } else {
//                bitSum = base;
//                base = 0;
//            }
//
//            if (header == null) {
//                header = new ListNode(bitSum);
//                current = header;
//            } else {
//                current.next = new ListNode(bitSum);
//                current = current.next;
//            }
//        }
//
//        return header;

        //官方标准答案
        ListNode dummyHead = new ListNode(0);
        ListNode p = l1, q = l2, curr = dummyHead;
        int carry = 0;
        while (p != null || q != null) {
            int x = (p != null) ? p.val : 0;
            int y = (q != null) ? q.val : 0;
            int sum = carry + x + y;
            carry = sum / 10;
            curr.next = new ListNode(sum % 10);
            curr = curr.next;
            if (p != null) p = p.next;
            if (q != null) q = q.next;
        }
        if (carry > 0) {
            curr.next = new ListNode(carry);
        }
        return dummyHead.next;
    }

    public ListNode addTwoNumbersWay(ListNode l1, ListNode l2) {
        if(l1==null) return l2;
        if(l2==null) return l1;
        ListNode ans = null;
        ListNode cur = null;
        boolean first = true;
        int carry = 0;
        while(l1 != null || l2 != null) {
            int sum = 0;
            if(l1 == null) {
                sum = l2.val + carry;
                carry = 0;
                l2 = l2.next;
            } else if(l2 == null) {
                sum = l1.val + carry;
                carry = 0;
                l1 = l1.next;
            } else {
                sum = l1.val + l2.val + carry;
                carry = 0;
                l1 = l1.next;
                l2 = l2.next;
            }
            if(sum >= 10) {
                sum = sum - 10;
                carry = 1;
            }

            if(first) {
                cur = new ListNode(sum);
                ans = cur;
                first = false;
            } else {
                cur.next = new ListNode(sum);
                cur = cur.next;
            }
        }

        if(carry == 1) {
            cur.next = new ListNode(1);
        }

        return ans;
    }

    /**
     * 提交时间 提交结果 执行用时 内存消耗 语言
     * 几秒前	通过	3 ms	44.9 MB	Java
     * 15 分钟前	通过	5 ms	45.6 MB	Java
     */
    @Test
    public void test() {
        Assert.assertTrue(
                ListNode.isEquals(
                        new ListNode(new int[]{7,0,8}),
                        this.addTwoNumbers(new ListNode(new int[]{2,4,3}),new ListNode(new int[]{5,6,4}))
                        )
        );
        Assert.assertTrue(
                ListNode.isEquals(
                        new ListNode(new int[]{0,1}),
                        this.addTwoNumbers(new ListNode(new int[]{5}),new ListNode(new int[]{5}))
                        )
        );
    }

    @Test
    public void test2() {
        Assert.assertTrue(
                ListNode.isEquals(
                        new ListNode(new int[]{7,0,8}),
                        this.addTwoNumbersWay(new ListNode(new int[]{2,4,3}),new ListNode(new int[]{5,6,4}))
                        )
        );
        Assert.assertTrue(
                ListNode.isEquals(
                        new ListNode(new int[]{0,1}),
                        this.addTwoNumbersWay(new ListNode(new int[]{5}),new ListNode(new int[]{5}))
                        )
        );
    }

}